from pydantic import BaseModel, Field


class InputMessage(BaseModel):
    prompt: str = Field(..., title="提示词", max_length=4000)
    style: str | None = Field(None, title="大模型名称", max_length=300)


class ReturnMessage(BaseModel):
    prompt: str = Field(..., title="提示词", max_length=4000)
    url: str = Field(..., title="地址", max_length=1000)
    task_id: str | None = Field(None, title="task_id", max_length=300)
